How to calculate electricity bill in Python